package com.dji.dao;

import com.dji.model.Drone;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 无人机数据访问对象，用于模拟数据存储
 */
public class DroneDao {
    private static final Map<String, Drone> drones = new HashMap<>();
    
    static {
        // 初始化一些示例无人机数据
        Drone drone1 = new Drone("Mavic Air 2");
        drone1.setLatitude(31.2304);
        drone1.setLongitude(121.4737);
        
        Drone drone2 = new Drone("Phantom 4 Pro");
        drone2.setLatitude(31.2312);
        drone2.setLongitude(121.4750);
        
        Drone drone3 = new Drone("Inspire 2");
        drone3.setLatitude(31.2320);
        drone3.setLongitude(121.4760);
        
        drones.put(drone1.getId(), drone1);
        drones.put(drone2.getId(), drone2);
        drones.put(drone3.getId(), drone3);
    }
    
    /**
     * 获取所有无人机列表
     */
    public List<Drone> getAllDrones() {
        return new ArrayList<>(drones.values());
    }
    
    /**
     * 根据ID获取无人机
     */
    public Drone getDroneById(String id) {
        return drones.get(id);
    }
    
    /**
     * 保存或更新无人机信息
     */
    public void saveDrone(Drone drone) {
        drones.put(drone.getId(), drone);
    }
    
    /**
     * 删除无人机
     */
    public void deleteDrone(String id) {
        drones.remove(id);
    }
    
    /**
     * 更新无人机状态
     */
    public void updateDroneStatus(String id, String status) {
        Drone drone = drones.get(id);
        if (drone != null) {
            drone.setStatus(status);
        }
    }
    
    /**
     * 更新无人机位置信息
     */
    public void updateDronePosition(String id, double latitude, double longitude, double altitude) {
        Drone drone = drones.get(id);
        if (drone != null) {
            drone.setLatitude(latitude);
            drone.setLongitude(longitude);
            drone.setAltitude(altitude);
        }
    }
}